AI-based diagnosis